2710c19025fcf679b2a8ac74a8d05ac48a017cfe,sdk/src/test/java/com/google/cloud/dataflow/sdk/io/bigtable/BigtableIOTest.java,BigtableIOTest,testReading,#,234

Before Change


    final int numRows = 1001;
    List<Row> testRows = makeTableData(table, numRows);

    TestPipeline p = TestPipeline.create();
    PCollection<Row> rows = p.apply(defaultRead.withTableId(table));
    DataflowAssert.that(rows).containsInAnyOrder(testRows);

    p.run();
    logged.verifyInfo(String.format("Closing reader after reading %d records.", numRows));
  }

After Change


    final int numRows = 1001;
    List<Row> testRows = makeTableData(table, numRows);

    runReadTest(defaultRead.withTableId(table), testRows);
    logged.verifyInfo(String.format("Closing reader after reading %d records.", numRows));
  }